#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@Project : light-edit
@File : ip_scan.py
@Author : Yj
@Time : 2025/9/9
@脚本说明 : IP 段扫描模块，检测存活主机
"""
import nmap


def scan_ip(ip_segment):
    """扫描指定 IP 段，检测存活主机"""
    nm = nmap.PortScanner()
    nm.scan(hosts=ip_segment, arguments='-sn')  # 主机发现扫描
    alive_hosts = []
    for host in nm.all_hosts():
        if nm[host].state() == 'up':
            print(f"[+] {host} 存活")
            alive_hosts.append(host)
    return alive_hosts
